988b475da808266f36f63898f43820c91bf2f471,dl4j-examples/src/main/java/org/deeplearning4j/examples/unsupervised/variational/VariationalAutoEncoderExample.java,VariationalAutoEncoderExample,main,#String[]#,36

Before Change


//        List<float[]> temp = new ArrayList<>();
//        temp.add(net.params().data().asFloat());
        for( int i=0; i<nEpochs; i++ ){
            net.fit(trainIter);
            trainIter.reset();
//            System.out.println(Arrays.toString(net.params().data().asFloat()));
//            temp.add(net.params().data().asFloat());

After Change


        for( int i=0; i<nEpochs; i++ ){
            while(trainIter.hasNext()){
                DataSet ds = trainIter.next();
                net.fit(ds);

                //Every N minibatches: collect the test set latent space values for later plotting
                if(iterationCount++ % plottingLatentSpaceEveryNMinibatches == 0){